潜在压缩与测试时训练:可训练 KV Cache 压缩
🧭 LLM 长上下文 KV cache 压缩 · 测试时训练 ★ 深度解读
Tsinghua University
将压缩 KV cache 视为可训练参数,用测试时训练拟合注意力与蒸馏,在 32× 压缩下 RULER 达 96.7%
📌 工作参考价值(4 点)
- 商品详情页、用户长期行为序列、长评论等重复使用的长上下文,可离线预压成可训练 KV cache。模型权重冻结,线上多 query 复用,适合推荐/搜索中大量候选文档场景;但注意论文只考核逻辑 KV-pair 预算,实际物理内存与速度需自行测评。
- 测试时训练不需要下游真实 query:用 480 个通用 prompt 与 teacher 全量 cache 生成 rollouts,就能提供 attention 与蒸馏监督。电商可以模板化商品属性、风格、摘要等通用问题,避免依赖真实用户日志。
- 初始化组合可直接复用:KVZip+ 重建分数选 preserver(占 75% 效果最好)、teacher attention 选 anchor、farthest-first 选 diversifier;ablations 表明 preserver 预算分配对准确率影响最大。
- 若把压缩文档 cache 与后续 query cache 合并,必须按 log-normalizer 加权而不是等权平均,否则准确率几乎归零;attention bias 贡献有限但可保留。
快手摘要注意力 V2 技术报告:C-KSA 与 S-KSA
🧭 长上下文高效注意力 · 二级KV压缩与稀疏索引 ★ 深度解读
Kuaishou
在 Hybrid KSA 上新增 C-KSA 与 S-KSA,把 Summary 同时用作可二级压缩 KV 记忆和稀疏索引,128K 下 KV cache 减少 40.4%、prefill 达 1.51×
📌 工作参考价值(4 点)
- 如果业务中已有或计划引入 Summary/landmark token 机制,可参考 C-KSA 的二级压缩:把每 km 个 Summary KV 池化为 layer-local Meta entry,Summary 序列继续按原粒度前向传播,远程缓存按 ks*km 倍压缩。这样可以从已有 Hybrid 模型低成本 retrofit,不用重新设计 chunk size。
- S-KSA 的 block-level routing 比 token-level 索引候选数少 b 倍,适合在电商搜索/广告 Agent 的长候选列表、长商品序列中做稀疏读取;其默认参数 b=4、remote budget K=4096、local window 1024 可作为起步配置,block size 放大到 8 会明显掉点,需谨慎。
- 如果只允许训练新模块,可冻结 backbone 做 module-only 适配:C-KSA 先对齐 attention 输出再全参微调,S-KSA 用 dense teacher 蒸馏 block 选择,18.5B tokens 即可接近 parent;block teacher 用 max pooling 不要用 sum pooling,后者会稀释 salient token 信号。
- 注意失败模式:频繁词提取等全局聚合任务在 128K 下从 45% 跌到 26.67%。若业务需要跨长上下文做类目频次、属性统计等聚合型召回/排序信号,应保留 dense 分支或额外聚合结构,不要完全依赖 sparse attention。
虚拟头注意力:线性重参数化实现双重 KV 压缩与质量提升
🧭 长上下文 LLM · 注意力重参数化 ★ 深度解读
Baidu Inc. · Peking University
通过 Q Premix 与 Linear Postmix 将训练后模型吸收为标准 GQA-2,实现 KV cache 4 倍压缩、训练 FLOPs 下降且质量提升
📌 工作参考价值(4 点)
- 线上长序列生成式推荐/Agent 工具调用受 KV cache 限制时,可参考 VHA:训练时加入线性头间混合,推理时重新参数化为标准 GQA-2,无需改推理框架;已有 GQA-8 服务可尝试降到 2 个 KV heads 并配合 Premix/Postmix,保持质量同时降低 4 倍缓存。
- (I+BA^T) 低秩 identity-residual 结构可迁移到多场景/多任务序列建模的输出融合,低秩约束提供隐式正则化,防止跨 head 混合过拟合;电商多业务场景可用其做 soft routing 而非共享 attention。
- Premix 的 per-group query transform 能实现 group-conditioned query specialization,在用户长期行为序列中根据不同 KV group 学习不同查询方向,不增加缓存即提升注意力多样性。
- 消融显示单纯 halve heads 会损失质量,必须同时在 query 侧做虚拟扩展、在输出侧做跨组融合;如果要在推荐模型中压缩 attention 头数,应同时设计两侧补偿,缺一不可。
注意力感知变换编码压缩 KV Cache
🧭 LLM 推理 · KV Cache 压缩 ★ 深度解读
Technical University of Munich · Technical University of Darmstadt · MIT · Harvard University · Red Hat AI
用白化变换与反向注水按注意力输出失真分配 KV cache bit,5.8x 压缩下近无损
📌 工作参考价值(4 点)
- 长上下文 LLM 推理服务里,KV cache 是大头成本;这套方法只需离线校准集做白化 + 全局 bit 分配,在线仅做 per-token 标量量化,无额外训练,适合快速接入 vLLM/TensorRT-LLM 的量化路径。
- 注意力感知权重可直接复用:key 通道用 q_c^2 加权,value 通道用 W_O 行范数加权;对重要通道多给 bit,尾部通道给 0 bit 等价低秩裁剪,比 uniform precision 更稳,尤其适合商品描述、多轮会话历史、Agent memory 等长上下文场景。
- 校准数据混合思路值得照搬:FineWeb 与 OpenR1-Math 各半,同时覆盖通用文本和推理链;电商场景可替换为商品描述/搜索会话/推理 trace 混合,避免只在通用语料上过拟合。
- 工程细节有复用价值:保留前 4 个 token 与最近 128 token 全精度,按 block 16 重量化;GQA 下按 head-group 做 token-wise max scaling;在更 aggressive 的 2-bit 预算下,attention-aware 分配相对 var-only 优势更明显,可做服务分层压缩。
页面局部紧凑键摘要用于高效长上下文解码
🧭 KV cache 压缩 · 稀疏注意力
通过页面局部光谱摘要和稀疏选择,大幅减少长上下文解码中 KV cache 读取量,保持质量
📌 工作参考价值(4 点)
- 电商推荐中常需用 LLM 建模长用户行为序列,LOCKS 的页面级压缩可将序列分块,每块保留紧凑谱摘要,在生成时仅注意 top 块,降低长序列推理延迟
- 该选择机制只基于摘要估计注意力质量,不读取原始键值,适合在推荐 query 生成等需要可控延迟的场景中部署
- 可作为 vLLM 插件直接集成到现有推理服务,无需修改模型,对已部署推荐系统的 LLM 组件改造成本极低
- 长形式推理(如 Agent 多步规划)收益最大,若推荐 Agent 需要长上下文推理,此方法可明显提升响应速度
HiKV:面向 LLM 解码的分层重要性感知 KV 缓存与硬件加速
🧭 LLM 推理 · KV 缓存压缩与硬件加速 ★ 深度解读
KU Leuven · UC Berkeley
分层 token 与元素双粒度重要性感知压缩,结合可重构硬件排序器,在 1% 精度损失内实现 7.95 倍加速和 90% 能耗降低
📌 工作参考价值(5 点)
- ## 业务可借鉴点
- 分层重要性思想可迁移至推荐系统稀疏化:HiKV 证明 KV 缓存冗余同时存在于 token 和元素两个粒度,建议在推荐模型(如 transformer-based)的长期行为序列建模中,尝试类似的「先筛重要交互、再筛每项交互中重要特征维度」的二阶段稀疏策略,用更少的存储/计算维持精度。
- 双银行 + 最小堆维护重要 token:Stage I 的 recent / important 银行与 min-heap 更新方式(HEAPIFY_UP/DOWN),可应用于 Agent 场景下的多轮对话历史管理:始终保留最近的若干交互,同时动态淘汰历史中重要性低的对话轮次,用 O(log B) 的代价保持固定预算。
- 分块并行排序近似全局 selection:Stage II 的 chunk-based sorting 可将 P 向量的 top-k 选择从 O(N log N) 降至 O(N log d_h),且召回率超 95%,可直接用于在线推理中需要选取重要 token/特征的场景(如长上下文中的关键 token 挑选),避免全局排序的高延迟。
- 硬件-算法协同设计视角:如果业务计划自研推理加速硬件,可参考「可重构重要性排序器(RIS)」的设计:同一电路通过配置支持堆操作和分块排序,面积仅增 8%。这提示在对延迟敏感的检索/推荐系统中,专用排序单元能以较低成本换取端到端效率。
可变速率训练无关KV缓存压缩用于长上下文大模型
🧭 长上下文LLM推理优化 · 可变速率KV缓存压缩
Iowa State University
提出无训练可变秩KV缓存压缩方法VarRate,按显著性分配预算并保留所有token,实现接近无损的长上下文推理
📌 工作参考价值(4 点)
- 在线LLM服务可直接部署VarRate作为无训练的KV缓存压缩插件,减少长上下文推理内存占用,支撑更长的用户历史(如推荐理由生成、Agent对话历史理解)。
- 在多查询复用预填充缓存的场景(如批量用户查询),VarRate避免丢弃token导致的精度崩溃,保障关键信息不丢失,提升推荐或搜索结果的稳定性。
- 显著性得分可基于注意力权重快速计算,适合流式处理,能集成到实时推理管线中,无需额外模型或微调,降低工程复杂度。
- 可变秩分配思想可推广到特征存储、用户画像压缩等模块,通过保留所有数据点但差异化精度,实现更好的资源效率平衡。
频率引导深度共享KV缓存压缩
🧭 长上下文LLM推理 · KV缓存压缩
Instituto de Investigación en Visión Artificial
推理时不需重训,通过频率分解与自适应模式分配压缩KV缓存,在长上下文任务中精度接近全缓存且实现3.9倍压缩
📌 工作参考价值(4 点)
- 对于利用长用户历史序列的推荐系统,可直接应用该技术压缩KV缓存,显著降低在线推理的显存和延迟
- 自适应分配压缩模式(共享深度/残差/精确)的思路可迁移到多任务推荐中的特征缓存管理,动态调整不同任务对历史状态的依赖
- 无需额外训练,适合快速集成到现有推理引擎,降低长上下文LLM在搜索、广告等场景的部署成本
- 低频分量共享策略可启发跨层特征融合设计,通过频域分解减少冗余缓存,提升工程效率
信息感知的 KV 缓存压缩用于长推理
🧭 LLM 推理优化 · 熵感知 KV 压缩
Shanghai Jiao Tong University · University of Edinburgh
提出熵感知的 KV 缓存压缩框架 InfoKV,结合预测不确定性与注意力得分,提升长推理效率。
📌 工作参考价值(4 点)
- 长对话/文档推理的部署加速:在电商搜索推荐系统的多轮助手、售后长文档分析等场景,可直接用 InfoKV 压缩 KV 缓存,减少显存占用与解码延迟,无需重新训练模型。
- Agent 长期记忆的高效管理:Agent 在维护对话历史或用户行为序列时,利用预测不确定性筛选对远期推理关键的信息 token,既保留关键上下文又控制缓存大小,提升持续交互的响应速度。
- 混合重要性评估范式:将 token 级熵信号与传统的注意力权重结合,是一种低成本、即插即用的重要性评分增强方式,可迁移到其它需要上下文剪枝的 LLM 应用中(如 RAG 中的长文档分块选择)。
- 解码阶段动态压缩:InfoKV 在解码阶段同样有效,适合流式生成时不断增长的 KV 缓存管理,可降低线上长时生成过程的尾部延迟。
KVServe:面向通信效率的拆解式LLM服务中服务感知KV缓存压缩
🧭 LLM推理优化 · KV缓存压缩 · 拆解式服务
University of Chinese Academy of Sciences · Institute of Computing Technology, Chinese Academy of Sciences · Shanghai Jiao Tong University
首个服务感知的自适应KV缓存压缩框架,通过贝叶斯搜索与在线控制实现动态策略选择,显著降低拆解式推理的通信延迟。
📌 工作参考价值(4 点)
- 电商/推荐系统若采用PD分离或KV拆解的LLM推理,可借鉴KVServe的“离线Pareto候选生成+在线bandit选择”模式,用轻量控制器动态匹配请求的时延/吞吐/质量约束,避免静态配置导致SLO违规。
- 自适应压缩思想可迁移到生成式推荐中的Embedding或特征传输压缩:根据网带宽、查询复杂度和延迟预算等上下文,动态切换量化/稀疏化/选择策略,而非固定一种压缩方法。
- 贝叶斯Profiling Engine通过全局探索+局部细搜+3D Pareto剪枝的方式大幅降低搜索开销(50×),在推荐系统的超参或策略组合爆炸场景中,可用于快速生成高质量候选配置集。
- 在线控制中的“分析模型+bandit”机制可迁移到Agent多智体通信优化:为Agent间的消息大小、压缩等级建立延迟预测模型,结合多臂老虎机上探索,在满足延迟目标下最大化通信效率。